package 算法.leetcode.array;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * 两数之和
 * @author lchenglong
 * @date 2020/11/30
 */
public class TwoNumsSum {

    /**
     * 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
     *
     * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。
     *
     * 来源：力扣（LeetCode）
     * 链接：https://leetcode-cn.com/problems/two-sum
     * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
     */

    public static void main(String[] args) {
        TwoNumsSum instance = new TwoNumsSum();
        int[] nums = new int[]{2,7,11,15};
        int target = 9;
        int[] results = instance.twoSum3(nums, target);
        System.out.println(Arrays.toString(results));
    }

    public int[] twoSum(int[] nums, int target) {
        int[] results = new int[2];
        HashMap<Integer,Integer> map = new HashMap();
        for (int i=0;i<nums.length;i++) {
            if (map.containsKey(nums[i])) {
                if (2 * nums[i] == target) {
                    // 如果当前值等于目标值的一半, 那么就是这两个数
                    results[0] = map.get(nums[i]);
                    results[1] = i;
                    return results;
                }
            } else {
                map.put(nums[i], i);
            }
        }
        for (int i=0;i<nums.length;i++) {
            if (map.get(target-nums[i])!=null){
                results[0] = i;
                results[1] = map.get(target-nums[i]);
            }
        }
        return results;
    }

    /**
     * 假定target为模，则我们所求结果中的两个数互为补数，我们利用补数和Map来解决这个问题，map里存的是补数和索引
     * 首先遍历数组，如果当前值在map里面存在，说明数组中是有其补数的，直接返回当前值的索引和补数的索引，
     * 否则，将当前值的补数作为key,索引作为value,存放到map中
     * @param nums
     * @param target
     * @return
     */
    public int[] twoSum2(int[] nums, int target) {
        int[] result = new int[2];
        Map<Integer,Integer> map = new HashMap(nums.length);
        for (int i=0;i<nums.length;i++){
            int source = target-nums[i];
            if (map.containsKey(nums[i])) {
                // 当前值在map中，说明其补数也在map中，所以得到想要的结果了
                result[0] = map.get(nums[i]);
                result[1] = i;
                return result;
            } else {
                // 将补数和索引放到map中
                map.put(source, i);
            }
        }
        return result;
    }

    public int[] twoSum3(int[] nums,int target) {
        int[] result = new int[2];
        Map<Integer,Integer> map = new HashMap<Integer, Integer>();
        for (int i=0,size = nums.length;i<size;i++) {
            int next = target - nums[i];
            if (map.containsKey(nums[i])) {
                // 如果当前值在map中，肯定是其target的补数放进去的
                result[0] = map.get(nums[i]) + 1;
                result[1] = i + 1 ;
                return result;
            } else {
                // 将补数和索引放入map中
                map.put(next,i);
            }
        }
        return result;
    }
}
